from pymongo import MongoClient


class DataBase(object):
    def __init__(self, db, table):
        client = MongoClient()
        db = client[db]
        self.collection = db[table]

    def get_collection(self):
        return self.collection